package targetToOffer.forteenDay;

/**
 * @author Godc
 * @description
 */
public class LowestCommonAncestor {
}
class Solution {
    /**
     * 前提排序二叉树
     * @param root
     * @param p
     * @param q
     * @return
     */
    public TreeNode lowestCommonAncestor(TreeNode root, TreeNode p, TreeNode q) {
        // 排序二叉树，如果pq在root两侧，说明root就是pq的最近祖先节点
        // 当pq都在左子树，那么递归左半边，否则递归右半边
        if(p.val>root.val&&q.val>root.val){
            return lowestCommonAncestor(root.right,p,q);
        }
        if(p.val<root.val&&q.val<root.val){
            return lowestCommonAncestor(root.left,p,q);
        }
        return root;
    }

    /**
     * 普通的二叉树
     * @param root
     * @param p
     * @param q
     * @return
     */
    public TreeNode lowestCommonAncestor1(TreeNode root, TreeNode p, TreeNode q) {
        // 越过子节点返回null，当 root等于 p,q，无需再向下遍历了，肯定不可能再pq之下
        if(root==null||p==root||q==root)
            return root;
        // 左右递归
        TreeNode left = lowestCommonAncestor(root.left, p, q);
        TreeNode right = lowestCommonAncestor(root.right, p, q);
        if(left==null&&right==null) return null;
        // 此时l和r都在root左侧，返回left,此时可能有两种情况，
        // p，q有一个在right中，那么我们返回的right指向p或q；p，q都在right中那么我们返回的就是最近公共父节点
        if(right==null) return left;
        if(left==null) return right;
        // 如果left和right都不为null，说明pq在root两侧，就是最近的祖先节点
        return root;
    }
}

 class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode(int x) { val = x; }
  }